# 初始化环境
import byzerllm
from real_agent.agent import Agent, Group

# 创建 LLM 实例
llm = byzerllm.ByzerLLM.from_default_model("deepseek_chat")

# 创建三个智能体
alice = Agent("Alice", llm)
bob = Agent("Bob", llm)
charlie = Agent("Charlie", llm)
charlie.when_to_refuse_reply("当用户回复感谢之类的客套话，直接选择ignore策略")

print("智能体创建完成")

# 创建开发群组
dev_group = Group("dev")
alice.join_group(dev_group)
# Bob 和 Charlie 加入群组
bob.join_group(dev_group)
charlie.join_group(dev_group)

# 私聊示例
alice.talk_to(dev_group, "能帮我写个快速排序算法么",
              mentions=[charlie],
              print_conversation=True)

# alice.talk_to(charlie, "能帮我写个快速排序算法么",
#               print_conversation=True)
